"""
配置文件，存储全局配置和状态
"""

# MySQL数据库配置
# 注意: 请替换以下信息为您的MySQL数据库配置
DB_CONFIG = {
    "host": "localhost",
    "user": "root",
    "password": "1219",  # 替换为您的MySQL密码
    "port": 3306,
    "database": "bank"  # 数据库名称
}

# 数据源标志
# True: 使用导入的数据文件 False: 使用已配置的MySQL数据库
USING_IMPORTED_DATA = False

# 全局状态
DB_SCHEMA = None
CHAT_HISTORY = []
QUERY_OPTIONS = {
    "allow_joins": False,
    "complex_aggregations": False,
    "max_results": 100
}

# 临时数据库表名前缀(用于导入数据)
TEMP_TABLE_PREFIX = "imported_"

# AI模型配置
# 注意：替换为您的API密钥
TONGYI_API_KEY = "sk-cbf498556ae24416b79ac7e14b5e90c8"
DEEPSEEK_API_KEY = "sk-6cf395bacf984f35a137d4dc2e259bd4"

# 使用的AI模型 - 'tongyi'、'deepseek' 或 'local'
ACTIVE_AI_MODEL = "deepseek"  # 默认使用通义千问

# AI模型选项
AI_MODELS = {
    "tongyi": "通义千问",
    "deepseek": "DeepSeek",
    "local": "本地模型"
}

# 提示词模板
PROMPT_TEMPLATE_IDENTITY = """
你是一个银行、财务和数据库领域专家，你可以回答用户的普通问题，也可以将自然语言问题转换为MySQL数据库的SQL查询。
"""

# 知识库配置
KNOWLEDGE_BASE = {
    "finance": {
        "path": "knowledge/finance",
        "description": "金融领域知识"
    },
    "banking": {
        "path": "knowledge/banking",
        "description": "银行业务知识"
    },
    "database": {
        "path": "knowledge/database",
        "description": "数据库领域知识"
    }
}

# 知识库嵌入模型配置
EMBEDDING_MODEL = {
    "model_name": "text2vec-base",
    "dimension": 768,
    "cache_dir": "embeddings_cache"
}

